# -*- coding: utf-8 -*-
# @Author  : gaoyu
# @Time    : 2024/9/25 3:52 PM
# @Function: 

import json
import os
from dotenv import load_dotenv
from loguru import logger

load_dotenv()

# Ollama 配置
# Ollama url
OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://10.48.0.81:11434")
# 常规使用的llm模型
LLM_MODEL = os.getenv("LLM_MODEL", "qwen2.5:32b")
# 标签生成的llm模型
TAG_LLM_MODEL = os.getenv("TAG_LLM_MODEL", "qwen2.5")
# 对话生成的llm模型
CHAT_LLM_MODEL = os.getenv("CHAT_LLM_MODEL", "qwen2.5")
# 代理生成的llm模型
AGENT_LLM_MODEL = os.getenv("AGENT_LLM_MODEL", "qwen2.5:14b")
# 代码生成的llm模型
CODE_LLM_MODEL = os.getenv("CODE_LLM_MODEL", "qwen2.5-coder")

# 数据库查询配置
VECTOR_DATABASE_URL = os.getenv("VECTOR_DATABASE_URL", "postgresql+psycopg://postgres:water@10.10.10.125:5432/langchain")
CACHE_DATABASE_URL = os.getenv("CACHE_DATABASE_URL", "postgresql://postgres:water@10.10.10.125:5432/langchain")
ORM_DATABASE_URL = os.getenv("ORM_DATABASE_URL", "postgresql://postgres:water@10.10.10.125:5432/langchain")
WATER_DATA_DATABASE_URL = os.getenv("WATER_DATA_DATABASE_URL", "postgresql://postgres:water@10.55.0.102:5432/water")
WATER_DATABASE_URL = os.getenv("WATER_DATABASE_URL", "postgresql://postgres:water@172.20.7.85:5432/water")






if __name__ == '__main__':

    logger.info(type(LLM_MODEL))
    logger.info(VECTOR_DATABASE_URL)
    logger.info(type(VECTOR_DATABASE_URL))
